"""
动态gdp柱状图绘制
"""
from pyecharts.charts import Bar,Timeline
from pyecharts.globals import ThemeType
from pyecharts.options import *
import json
#打开文件
f = open("C:/Users/86131/Desktop/课程/python资料/资料/第1-12章资料/资料/可视化案例数据/动态柱状图数据/1960-2019全球GDP数据.csv","r",encoding = "GB2312")
data_lines = f.readlines()
#关闭文件
f.close()
#数据的预处理,删除第一行
data_lines.pop(0)

#将数据格式转化为字典!!!
#{年份:[[国家,gdp],[国家,gdp],...]],年份:[[国家,gdp],[国家,gdp],...]],年份:[[国家,gdp],[国家,gdp],...]]...}
#{1960:[[美国,123],[中国,321],...]],1961:[[国家,gdp],[国家,gdp],...]],年份:[[国家,gdp],[国家,gdp],...]]...}

data_dict = {}
for line in data_lines:
    year = int(line.split(",")[0])
    country = line.split(",")[1]
    gdp = float(line.split(",")[2])
    try:#抛出一个异常,如果字典中还没好友对应的year值,就进行追加操作会报错
        data_dict[year].append([country,gdp])
    except Exception as e:
        data_dict[year] = []
        data_dict[year].append([country, gdp])
#创建一个时间线对象
timeline = Timeline({"theme": ThemeType.LIGHT})
#排序年份
sorted_year_list = sorted(data_dict.keys())
for year in sorted_year_list:
    data_dict[year].sort(key = lambda element: element[1], reverse = True)#翻转一下谁的gdp高谁占第一位
    year_data = data_dict[year][:8]#取出本年排名前八的国家
    x_data = []#初始化x轴数据列表(国家
    y_data = []#初始化y轴数据列表(gdp
    for country_gdp in year_data:
        x_data.append(country_gdp[0])#x轴添加国家
        y_data.append(country_gdp[1]/100000000)#y轴添加GDP

    #构建柱状表,每一年都创建一个柱状图
    bar = Bar()

    x_data.reverse()
    y_data.reverse()
    bar.add_xaxis(x_data)
    bar.add_yaxis("GDP(亿)",y_data,label_opts=LabelOpts(position="right"))
    #反转x轴和y轴
    bar.reversal_axis()
    bar.set_global_opts(
        title_opts=TitleOpts(title=f"{year}年全球前8GDP数据的国家")
    )
    timeline.add(bar,str(year))#传入每一年,和这个柱状图

#创建时间线对象

#设置时间线自动播放
timeline.add_schema(
    play_interval=1000,
    is_timeline_show=True,
    is_auto_play=True,
    is_loop_play=True
)

#绘制图像
timeline.render("1960-2019年全球gdp前八的国家.html")












